num_workers=4
root_dir=merge
batch_size=5
lr=1e-4
epochs=100
net_G=segformer
last_epoch=-1
checkpoint=pretrained/mit_b3.pth
do=train
in_channels=3

python main.py \
    --num_workers=$num_workers \
    --root_dir=$root_dir \
    --batch_size=$batch_size \
    --lr=$lr \
    --epochs=$epochs \
    --net_G=$net_G \
    --last_epoch=$last_epoch \
    --checkpoint=$checkpoint \
    --do=$do \
    --in_channels=$in_channels